package cn.tedu.article.controller;

import cn.tedu.base.response.JsonResult;
import cn.tedu.base.response.StatusCode;
import io.swagger.annotations.Api;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.support.ByteArrayMultipartFileEditor;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

@Slf4j
@RestController
@RequestMapping("/file")
@Api(tags = "文件上传模块")
public class UploadController {
    /**
     * 定义处理图片上传的方法
     * MultipartFile用来接收上传图片文件的对象
     * @param file 形参file必须与前端upload上传组件的name一致!
     * @return
     */
    @PostMapping("/upload")
    public JsonResult upload(MultipartFile file) throws IOException {
        //1.获取原始文件名
        String fileName = file.getOriginalFilename();
        log.debug("文件名: "+fileName);
        //2.截取文件后缀名
//        String suffix = fileName.substring(fileName.lastIndexOf("."));
        String suffix = ".png";
        //3.生成一个不会重复的随机字符串作为新的文件名
        fileName = UUID.randomUUID()+suffix;
        //4.指定一个路径用于存储上传到文件图片，注意！这个路径必须自己电脑上有！
        String dirPath = "d:/files";
        //5.为了提高检索性能，我们可以设置一个自定义的日期路径作为图片存储的文件夹
        SimpleDateFormat sdf = new SimpleDateFormat("/yyyy/MM/dd/");
        //6.将当前日期转为上面的格式
        String datePath = sdf.format(new Date());
        //7.将上面的盘符路径于日期路径拼接成一个完整的文件夹路径，用于存储图片
        File dirFile = new File(dirPath+datePath);
        //8.如果该对象对应的文件夹不存在，则依照此文件夹路径创建多层文件夹
        if(!dirFile.exists()){
            dirFile.mkdirs();
        }
        //9.拼接一个完整的图片路径
        String filePath = dirPath+datePath+fileName;
        //10.将上传的图片文件存储到指定路径
        file.transferTo(new File(filePath));

        //生成缩略图开始,通过工具类得到二进制数组
        byte[] bytes = ThumbnailGenerator.generateThumbnail(filePath, 100, 100);
        //指定缩略图存储路径
        String thumbnailUrl = "d:/files/thumbnail" + datePath + fileName;
        //创建缩略图文件夹
        File thumbnailDirFile = new File("d:/files/thumbnail" + datePath);
        if(!thumbnailDirFile.exists()){
            thumbnailDirFile.mkdirs();
        }
        //将缩略图保存到指定路径
        saveBytesToFile(bytes,thumbnailUrl);

        return new JsonResult(StatusCode.OPERATION_SUCCESS,dirPath+datePath+fileName);
    }

    @PostMapping("remove")
    public JsonResult remove(String imgUrl){
        log.debug("删除图片:",imgUrl);
        new File("d:/files"+imgUrl).delete();
        return JsonResult.ok(StatusCode.SUCCESS);
    }

    private static void saveBytesToFile(byte[] bytes, String filePath) {
        try (FileOutputStream fos = new FileOutputStream(filePath)) {
            fos.write(bytes);
            System.out.println("缩略图已成功保存到: " + filePath);
        } catch (IOException e) {
            e.printStackTrace();
            System.err.println("保存缩略图时发生错误: " + e.getMessage());
        }
    }

}